77a3189d40bfcac007d99ac8b9f888dbae32ad07,python/src/com/jetbrains/python/refactoring/surround/surrounders/statements/PyWithIfSurrounder.java,PyWithIfSurrounder,surroundStatement,#Project#Editor#PsiElement[]#,25

Before Change


    PyIfStatement ifStatement =
      PythonLanguage.getInstance().getElementGenerator().createFromText(project, PyIfStatement.class, "if True:\n    ");
    final PsiElement parent = elements[0].getParent();
    ifStatement.getIfPart().addRange(elements[0], elements[elements.length - 1]);
    ifStatement = (PyIfStatement) parent.addBefore(ifStatement, elements[0]);
    parent.deleteChildRange(elements[0], elements[elements.length - 1]);

After Change


    PyIfStatement ifStatement =
      PythonLanguage.getInstance().getElementGenerator().createFromText(project, PyIfStatement.class, "if True:\n    ");
    final PsiElement parent = elements[0].getParent();
    final PyStatementList statementList = ifStatement.getIfPart().getStatementList();
    assert statementList != null;
    statementList.addRange(elements[0], elements[elements.length - 1]);
    ifStatement = (PyIfStatement) parent.addBefore(ifStatement, elements[0]);